List comprehension in list within a list (dimensions are different than usual)